import base64
from Crypto.Cipher import AES


def add_to_16(value):  # 不足16位补满16位。
    while len(value) % 16 != 0:
        value += '\0'
    return str.encode(value)  # 返回bytes


def encrypt_oracle(password):
    key = 'lab202'  # 秘钥
    text = password # 待加密文本
    aes = AES.new(add_to_16(key), AES.MODE_ECB) # 先进行aes加密
    encrypt_aes = aes.encrypt(add_to_16(text)) # 用base64转成字符串形式
    encrypted_text = str(base64.encodebytes(encrypt_aes), encoding='utf-8') # 执行加密并转码返回bytes
#    print(encrypted_text)
    return encrypted_text  # 返回加密的数据
    # 解密方法


def decrypt_oralce(password):
    # 秘钥
    key = 'lab202'
    # 密文
    text = password
    # 初始化加密器
    aes = AES.new(add_to_16(key), AES.MODE_ECB)
    # 优先逆向解密base64成 bytes
    base64_decrypted = base64.decodebytes(text.encode(encoding='utf-8')) #执行解密密并转码返回
    decrypted_text = str(aes.decrypt(base64_decrypted),encoding='utf-8').replace('\0','')
#    print(text+':'+decrypted_text)
    return decrypted_text
